import os
from pypdf import PdfWriter, PdfReader
import time



def cheak_folder():
    try:
        if os.path.isdir('input') == True:
            pass
        else:
            os.mkdir("input")
            print('Папка ввода отсутствует')
        if os.path.isdir('output') == True:
            pass
        else:
            print('Папка вывода отсутствует')
            os.mkdir("output")
        if os.path.isfile('bg.pdf') == True:
             pass
        else:
            print('Файл печат отсутствует')
            
    except:
        print('Папка ввода не найдена')


def start():
    all_files = []
    all_files = os.listdir('input')
    stamp = PdfReader("bg.pdf").pages[0]
    for files in all_files:
            writer = PdfWriter(clone_from=f"input/{files}")
            writer.pages[-1].merge_page(stamp)
            files = 'out_' + files 
            writer.write(f"output/{files}")

def main():
    cheak_folder()
    time.sleep(1)
    start()


main()

